# 1. 初始化浏览器（以 Chrome 为例）
from selenium.webdriver import Keys
from selenium.webdriver.chrome import webdriver
from selenium.webdriver.common.by import By

from utils.driver_utils import get_driver

driver = get_driver()
driver.get("https://www.baidu.com")  # 打开百度首页

# 2. 定位搜索输入框（百度搜索框的 id 是 "kw"）
search_input = driver.find_element(By.ID, "kw")

# 3. 输入内容 + 按回车（两种写法均可）
# 写法 1：先输入，再单独按回车
search_input.send_keys("Selenium 回车操作")  # 输入内容
search_input.send_keys(Keys.ENTER)  # 按回车

# 写法 2：输入内容后直接拼接回车（更简洁）
# search_input.send_keys("Selenium 回车操作" + Keys.ENTER)

# 4. 等待结果加载（可选，避免页面未加载完就关闭）
import time
time.sleep(3)

# 5. 关闭浏览器
driver.quit()